Data Structures using python